package com.example.javarabbitmq;

import com.rabbitmq.client.*;

import java.io.IOException;
import java.util.HashMap;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

/**
 * RabbitMQ消费者
 * @author HAN Bing
 * @date 2021/5/29
 */
public class RabbitConsumer {
    private static final String QUEUE_NAME = "queue_demo";
    private static final String IP_ADDRESS = "8.131.233.42";
    private static final int PORT = 5672;

    public static void main(String[] args) throws IOException, TimeoutException, InterruptedException {
        Address[] addresses = new Address[]{new Address(IP_ADDRESS, PORT)};
        ConnectionFactory factory = new ConnectionFactory();
        factory.setUsername("root");
        factory.setPassword("d");
        // 这里的连接方式与生产者的demo略有不同，注意辨别
        // 创建连接
        Connection connection = factory.newConnection(addresses);
        // 创建信道
        final Channel channel = connection.createChannel();
        // 设置客户端最多接收未被ack的消息的个数
        channel.basicQos(64);

        Consumer consumer = new DefaultConsumer(channel) {
            @Override
            public void handleDelivery(String consumerTag,
                                       Envelope envelope,
                                       AMQP.BasicProperties properties,
                                       byte[] body) throws IOException {
                System.out.println("recv message: " + new String(body));
                try {
                    TimeUnit.SECONDS.sleep(1);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                // 显式确认可以防止消息丢失，multiple表示此tag编号之前的所有
                //                                           非multiple
                channel.basicAck(envelope.getDeliveryTag(), false);
            }
        };
        // b1是noLocal，设置为true则表示不能将同一个Connection中生产者发送的消息传送给这个Connection中的消费者
        // b2是exclusive
        //                      队列名    是否自动确认 consumerTag   不给同一个connection中的消费者     是否排他      其他参数        回调函数
        channel.basicConsume(QUEUE_NAME, false,    "",            false,            false, new HashMap<>(), consumer);

//        // 拉模式
//        //                                                  非自动确认
//        GetResponse response = channel.basicGet(QUEUE_NAME, false);
//        channel.basicAck(response.getEnvelope().getDeliveryTag(), false);

        // 等待回调函数执行完毕之后，关闭资源
        TimeUnit.SECONDS.sleep(5);
        channel.close();
        connection.close();
    }
}
